package com.tangyuan.ai.entity.enums;


import lombok.AllArgsConstructor;
import lombok.Getter;

@Getter
@AllArgsConstructor
public enum ModelTypeEnum {
    ERNIE_4(1,"ERNIE-4.0-8K"),
    AI_DRAWING(2,"AI_DRAWING"),
    ;

    public final Integer id;
    public final String modelName;


    public static ModelTypeEnum getModelTypeEnum(Integer id) {
        for (ModelTypeEnum value : ModelTypeEnum.values()) {
            if(id.equals(value.getId())){
                return value;
            }
        }
        return ModelTypeEnum.ERNIE_4;
    }

}
